shared source/vb/language/shared/ccomptrex.h

   1 //-------------------------------------------------------------------------------------------------
   2 //
   3 //  Copyright (c) Microsoft Corporation.  All rights reserved.
   4 //
   5 //  Wrapper for CComPtrEx which can handle inheritance scenarios where there are multiple paths
   6 //  to IUnknown
   7 //
   8 //-------------------------------------------------------------------------------------------------
   9
  10 #pragma once
  11
  12 template <class T>
  13 class CComPtrEx : public CComPtrBase<T>
  14 {
  15 public:
  16     CComPtrEx() throw()
  17     {
  18     }
  19     CComPtrEx(T* lp) throw() :
  20         CComPtrBase<T>(lp)
  21
  22     {
  23     }
  24     CComPtrEx(_In_ const CComPtrEx<T>& lp) throw() :
  25         CComPtrBase<T>(lp.p)
  26     {
  27     }
  28     T* operator=(_In_opt_ T* lp) throw()
  29     {
  30         if(*this!=lp)
  31         {
  32             CComPtrEx<T> sp(lp);
  33             TemplateUtil::Swap(&p, &sp.p);
  34         }
  35         return *this;
  36     }
  37     T* operator=(_In_ const CComPtrEx<T>& lp) throw()
  38     {
  39         if(*this!=lp)
  40         {
  41             CComPtrEx<T> sp(lp);
  42             TemplateUtil::Swap(&p, &sp.p);
  43         }
  44         return *this;
  45     }
  46 };
  47